package com.carrental.search.repository;

import com.carrental.search.model.document.VehicleDocument;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;

import java.math.BigDecimal;
import java.util.List;
import java.util.Optional;

/**
 * 车辆索引文档的基础存储库接口
 * 定义所有需要的方法，不依赖于Elasticsearch
 */
public interface VehicleRepositoryBase {

    /**
     * 根据车辆品牌查询
     *
     * @param brand 车辆品牌
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByBrand(String brand);

    /**
     * 根据车辆型号查询
     *
     * @param model 车辆型号
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByModel(String model);

    /**
     * 根据车辆类型查询
     *
     * @param type 车辆类型
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByType(String type);

    /**
     * 根据门店ID查询
     *
     * @param storeId 门店ID
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByStoreId(Long storeId);

    /**
     * 根据车辆状态查询
     *
     * @param status 车辆状态（0-维护中，1-可用，2-已租出）
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByStatus(Integer status);

    /**
     * 根据日租金范围查询
     *
     * @param minDailyRent 最小日租金
     * @param maxDailyRent 最大日租金
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByDailyRentBetween(BigDecimal minDailyRent, BigDecimal maxDailyRent);

    /**
     * 根据座位数查询
     *
     * @param seats 座位数
     * @return 车辆文档列表
     */
    List<VehicleDocument> findBySeats(Integer seats);

    /**
     * 根据车辆名称、品牌或型号模糊查询
     *
     * @param keyword 关键词
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByKeyword(String keyword);

    /**
     * 根据是否推荐查询
     *
     * @param recommended 是否推荐
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByRecommended(Boolean recommended);

    /**
     * 根据评分范围查询
     *
     * @param minRating 最小评分
     * @param maxRating 最大评分
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByRatingBetween(Float minRating, Float maxRating);

    /**
     * 根据生产年份范围查询
     *
     * @param minYear 最小年份
     * @param maxYear 最大年份
     * @return 车辆文档列表
     */
    List<VehicleDocument> findByYearBetween(Integer minYear, Integer maxYear);

    /**
     * 保存单个文档
     *
     * @param entity 要保存的文档
     * @return 保存后的文档
     */
    <S extends VehicleDocument> S save(S entity);

    /**
     * 批量保存文档
     *
     * @param entities 要保存的文档列表
     * @return 保存后的文档列表
     */
    <S extends VehicleDocument> Iterable<S> saveAll(Iterable<S> entities);

    /**
     * 根据ID查询文档
     *
     * @param id 文档ID
     * @return 文档
     */
    Optional<VehicleDocument> findById(Long id);

    /**
     * 检查文档是否存在
     *
     * @param id 文档ID
     * @return 是否存在
     */
    boolean existsById(Long id);

    /**
     * 查询所有文档
     *
     * @return 所有文档
     */
    Iterable<VehicleDocument> findAll();

    /**
     * 查询所有文档，并按指定规则排序
     *
     * @param sort 排序规则
     * @return 所有文档
     */
    Iterable<VehicleDocument> findAll(Sort sort);

    /**
     * 分页查询所有文档
     *
     * @param pageable 分页参数
     * @return 分页结果
     */
    Page<VehicleDocument> findAll(Pageable pageable);

    /**
     * 根据ID列表查询文档
     *
     * @param ids ID列表
     * @return 文档列表
     */
    Iterable<VehicleDocument> findAllById(Iterable<Long> ids);

    /**
     * 统计文档数量
     *
     * @return 文档数量
     */
    long count();

    /**
     * 根据ID删除文档
     *
     * @param id 文档ID
     */
    void deleteById(Long id);

    /**
     * 删除文档
     *
     * @param entity 要删除的文档
     */
    void delete(VehicleDocument entity);

    /**
     * 根据ID列表批量删除文档
     *
     * @param ids ID列表
     */
    void deleteAllById(Iterable<? extends Long> ids);

    /**
     * 批量删除文档
     *
     * @param entities 要删除的文档列表
     */
    void deleteAll(Iterable<? extends VehicleDocument> entities);

    /**
     * 删除所有文档
     */
    void deleteAll();

    /**
     * 查找相似文档
     *
     * @param entity 参考文档
     * @param fields 要比较的字段
     * @param pageable 分页参数
     * @return 相似文档列表
     */
    Page<VehicleDocument> searchSimilar(VehicleDocument entity, String[] fields, Pageable pageable);
} 